import os
from videoProcess.video import generate_video, generate_video_enhancement


def generate_dataset(path):
    """

    :param path:  目录
    :return:
    """
    i = 0
    for c_root, c_dirs, c_files in os.walk(path):
        for file in c_files:
            name = os.path.join(c_root, file)
            if name.find("Talking&Yawning") != -1:
                continue
            i = i + 1
            print(i)
            print(name)
            generate_video(name)


def generate_dataset_enhancement(path):
    """

    :param path:  目录
    :return:
    """
    i = 0
    for c_root, c_dirs, c_files in os.walk(path):
        for file in c_files:
            name = os.path.join(c_root, file)
            generate_video_enhancement(name)


if __name__ == "__main__":
    # generate_dataset(r"D:\dataset\YawDDdata_original\Mirror\Female_mirror")
    # generate_dataset(r"D:\dataset\YawDDdata_original\Mirror\Male_mirror")
    generate_dataset_enhancement(r"D:\dataset\YawDDdata_original\Mirror\all_cut")
